package com.birdboot.http;

import javax.activation.MimetypesFileTypeMap;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

/**
 * 响应对象
 * 该类的每一个实例用于表示HTTP协议规定的响应
 * 一个响应由三部分构成
 * 状态行，响应头，响应正文
 */
public class HttpServletResponse {
    private static MimetypesFileTypeMap mftm = new MimetypesFileTypeMap();

    private Socket socket;

    //状态行相关信息
    private int statusCode;//状态代码
    private String statusReason;//状态描述

    //响应头相关信息
    //key:响应头的名字  value:响应头对应的值
    private Map<String,String> headers = new HashMap<>();

    //响应正文相关信息
    private File contentFile;//正文对应的实体文件


    public HttpServletResponse(Socket socket){
        this.socket = socket;
    }

    /**
     * 响应方法
     * 该方法会将当前响应对象中的内容按照标准的HTTP响应格式发送给客户端
     */
    public void response() throws IOException {
        //1发送状态行
        sendStatusLine();
        //2发送响应头
        sendHeaders();
        //3发送响应正文
        sendContent();
    }
    //发送状态行
    private void sendStatusLine() throws IOException {
        println("HTTP/1.1"+" "+statusCode+" "+statusReason);
    }
    //发送响应头
    private void sendHeaders() throws IOException {
        //遍历headers将所有待发送的响应头发送给客户端
        Set<Map.Entry<String,String>> entrySet = headers.entrySet();
        for(Map.Entry<String,String> e :entrySet){
            /*
                headers
                key                 value
                Content-Type        text/html           Entry
                Content-Length      245                 Entry
                Server              BirdWebServer       Entry
             */
            String key = e.getKey();//响应头名字
            String value = e.getValue();//响应头对应的值
            println(key + ": " + value);
        }


        //单独发送回车+换行表达响应头部分发送完毕了
        println("");
    }
    //发送响应正文
    private void sendContent() throws IOException {
        if(contentFile!=null) {//包含正文时才发送正文文件
            OutputStream out = socket.getOutputStream();
            FileInputStream fis = new FileInputStream(contentFile);
            int len = 0;
            byte[] buf = new byte[1024 * 10];
            while ((len = fis.read(buf)) != -1) {
                out.write(buf, 0, len);
            }
        }
    }

    /**
     * 要求客户端重定向到指定位置
     * @param uri
     */
    public void sendRedirect(String uri){
        //1设置状态代码302
        setStatusCode(302);
        setStatusReason("Moved Temporarily");
        //2添加响应头Location
        addHeader("Location",uri);
    }


    /**
     * 向客户端发送一行字符串
     * @param line
     */
    private void println(String line) throws IOException {
        OutputStream out = socket.getOutputStream();
        byte[] data = line.getBytes(StandardCharsets.ISO_8859_1);
        out.write(data);
        out.write(13);
        out.write(10);
    }

    public int getStatusCode() {
        return statusCode;
    }

    public void setStatusCode(int statusCode) {
        this.statusCode = statusCode;
    }

    public String getStatusReason() {
        return statusReason;
    }

    public void setStatusReason(String statusReason) {
        this.statusReason = statusReason;
    }

    public File getContentFile() {
        return contentFile;
    }

    /**
     * 添加正文文件
     * 在添加文件的同时会自动根据该文件添加对应响应头:Content-Type和Content-Length
     * @param contentFile
     */
    public void setContentFile(File contentFile) {
        this.contentFile = contentFile;
        addHeader("Content-Type",mftm.getContentType(contentFile));
        addHeader("Content-Length",contentFile.length()+"");
    }

    /**
     * 添加一个响应头
     * @param name
     * @param value
     */
    public void addHeader(String name,String value){
        headers.put(name,value);
    }
}






